"""
数据操作类：读取测试数据
类名：DataOperation
方法：
    init：根据文件类型读取数据
    read_list：提取为列表数据
    read_dict：提取为字典数据
"""
import os
import pandas as pd
import os
from pprint import pprint

class DataOperation:
    def __init__(self, filename, sheet_name=0):
        _, extension = os.path.splitext(filename)
        # 构建文件路径
        file_path = os.path.join(os.path.abspath('data'), filename)

        # 读取文件
        if extension == '.csv':
            self.data = pd.read_csv(file_path)
        else:
            self.data = pd.read_excel(file_path, sheet_name)

    def read_list(self):
        return self.data.values.tolist()

    def read_dict(self):
        return [self.data.iloc[i].to_dict() for i in self.data.index.values.tolist()]


if __name__ == '__main__':
    data = DataOperation('user_info.csv')

    pprint(data.read_dict())
